package com.hliushi.nio.c4;

import lombok.extern.slf4j.Slf4j;

import java.io.IOException;
import java.net.InetSocketAddress;
import java.nio.ByteBuffer;
import java.nio.channels.ServerSocketChannel;
import java.nio.channels.SocketChannel;
import java.util.ArrayList;
import java.util.List;

import static com.hliushi.netty.util.ByteBufferUtil.debugRead;

/**
 * @author llhuang10
 * @date 2021/10/17 17:30
 */
@Slf4j
public class Server {
    public static void main(String[] args) throws IOException, InterruptedException {
        // 单线程 + 阻塞(事件没有发生, 线程将暂停, 一个方法会阻塞另一个方法的调用)
        // blockMode();

        // 单线程 + 非阻塞(事件没有发生, 线程不会停, 继续往下运行)
        // 可以处理多个客户端的连接/读/写请求. 但线程一直运行, CPU空转, 浪费CPU资源
        // 改进方法, 当没有事件的时候, 让线程休息 -> 避免CPU空转
        nonBlockMode();
    }

    private static void nonBlockMode() throws IOException, InterruptedException {
        // 使用 nio 来理解阻塞模式
        // 单线程来处理, 看看能不能处理这么多连接
        // 阻塞在哪里    1是连接的阻塞 2是读取数据的阻塞

        // 0.ByteBuffer
        ByteBuffer buffer = ByteBuffer.allocate(16);
        // 1.创建了服务器
        ServerSocketChannel ssc = ServerSocketChannel.open();

        // false: ssc非阻塞模式 -> ssc.accept() -> 非阻塞, 线程还会继续运行, CPU一直运行, 如果没有建立连接, 但sc为null
        ssc.configureBlocking(false);
        // 2.绑定监听端口
        ssc.bind(new InetSocketAddress(10010));

        // 3.连接集合
        List<SocketChannel> channels = new ArrayList<>();

        while (true) {
            // 4.accept建立与客户端连接, SocketChannel用户与客户端之间通信
            // log.debug("connecting..."); // 非阻塞注释掉防止刷屏
            // 默认阻塞方法, 线程停止运行, ssc.configureBlocking(false)设置为非阻塞
            SocketChannel sc = ssc.accept();
            if (sc != null) {
                log.debug("connected... {}", sc);
                // false: sc非阻塞模式 -> channel.read() -> 非阻塞, 线程仍会继续运行, 如果没有读取到数据, read返回 0
                sc.configureBlocking(false);
                channels.add(sc);
            }

            for (SocketChannel channel : channels) {
                // 5.接收客户端发送的数据
                // log.debug("before read... {}", channel); // 非阻塞注释掉防止刷屏
                // 默认阻塞方法, sc.configureBlocking(false)设置为非阻塞
                int read = channel.read(buffer);
                if (read > 0) {
                    buffer.flip();
                    debugRead(buffer);
                    buffer.clear();
                    log.debug("after read... {}", channel);
                }
            }
            Thread.sleep(10);
            // channels.forEach(channel -> {
            //     // 5.接收客户端发送的数据
            //     // log.debug("before read... {}", channel); // 非阻塞注释掉防止刷屏
            //     try {
            //         // 默认阻塞方法, sc.configureBlocking(false)设置为非阻塞
            //         int read = channel.read(buffer);
            //         if (read > 0) {
            //             buffer.flip();
            //             debugRead(buffer);
            //             buffer.clear();
            //             log.debug("after read... {}", channel);
            //         }
            //     } catch (IOException e) {
            //         e.printStackTrace();
            //     }
            // });
        }
    }

    private static void blockMode() throws IOException {
        // 0.ByteBuffer
        ByteBuffer buffer = ByteBuffer.allocate(16);
        // 1.创建了服务器
        ServerSocketChannel ssc = ServerSocketChannel.open();
        // false: 非阻塞模式 -> ssc.accept() -> 非阻塞, 线程还会继续运行, CPU一直运行, 如果没有建立连接, 但sc为null
        // true:  阻塞模式, 默认
        ssc.configureBlocking(true);
        ssc.bind(new InetSocketAddress(10010));
        // 3.连接集合
        List<SocketChannel> channels = new ArrayList<>();

        while (true) {
            // 4.accept建立与客户端连接, SocketChannel用户与客户端之间通信
            log.debug("connecting...");
            SocketChannel sc = ssc.accept();    // 默认阻塞方法, 线程停止运行
            log.debug("connected... {}", sc);
            if (sc != null) {
                // false: 非阻塞模式 -> channel.read() -> 非阻塞, 线程仍会继续运行, 如果没有读取到数据, read返回 0
                // true:  阻塞模式, 默认
                sc.configureBlocking(true);
                channels.add(sc);
            }

            for (SocketChannel channel : channels) {
                // 5.接收客户端发送的数据
                log.debug("before read... {}", channel);
                // 默认阻塞方法, sc.configureBlocking(false)设置为非阻塞
                int read = channel.read(buffer);
                if (read > 0) {
                    buffer.flip();
                    debugRead(buffer);
                    buffer.clear();
                    log.debug("after read... {}", channel);
                }
            }
        }
    }
}
